#ifndef GRASP_NMS_H
#define GRASP_NMS_H

#include <torch/serialize/tensor.h>
#include <vector>
#include <cuda.h>
#include <cuda_runtime_api.h>

int grasp_nms_gpu(at::Tensor grasps, at::Tensor keep, float nms_metric_thresh);
int grasp_distance_gpu(at::Tensor grasps_a, at::Tensor grasps_b, at::Tensor ans_distance);
int roipool3d_gpu(at::Tensor xyz, at::Tensor boxes3d, at::Tensor pts_feature, at::Tensor pooled_features, at::Tensor pooled_empty_flag);

#endif
